python list comprehension vs for loop